import sys
import os

def check_environment():
    """检查运行环境"""
    print("=== 环境检查 ===")
    print(f"Python版本: {sys.version}")
    print(f"平台: {sys.platform}")

    # 检查关键库
    libraries = [
        'torch', 'transformers', 'soundfile',
        'librosa', 'numpy', 'accelerate'
    ]

    print("\n=== 库版本检查 ===")
    for lib in libraries:
        try:
            module = __import__(lib)
            version = getattr(module, '__version__', 'Unknown')
            print(f"✓ {lib}: {version}")
        except ImportError:
            print(f"✗ {lib}: 未安装")

    # 检查CUDA
    try:
        import torch
        if torch.cuda.is_available():
            print(f"\n✓ CUDA可用: {torch.cuda.get_device_name(0)}")
            print(f"显存: {torch.cuda.get_device_properties(0).total_memory / 1024 ** 3:.1f} GB")
        else:
            print("\n⚠ CUDA不可用，将使用CPU")
    except:
        print("\n⚠ 无法检查CUDA状态")

    # 检查文件路径
    test_file = r"C:\Users\马军\Desktop\pycharm-workspace\model-study\seamless_m4t\G.E.M.邓紫棋 - 句号.mp3"
    print(f"\n=== 文件检查 ===")
    if os.path.exists(test_file):
        print(f"✓ 音频文件存在: {test_file}")
        print(f"文件大小: {os.path.getsize(test_file) / 1024 ** 2:.1f} MB")
    else:
        print(f"✗ 音频文件不存在: {test_file}")

    print("\n=== 建议 ===")
    if sys.version_info >= (3, 13):
        print("⚠ Python 3.13可能存在兼容性问题，建议使用Python 3.9-3.11")

    print("如果遇到numpy警告，执行: pip install numpy==1.24.3")
    print("如果模型加载慢，确保网络连接正常")


if __name__ == "__main__":
    check_environment()